import os
import math
import requests
from tqdm import tqdm

# --- 1. 配置区域 ---
# 你可以修改这里的参数来定义下载范围

# 下载的缩放层级 (z)。层级越高，细节越丰富，文件也越多。
# 建议先从较小的层级（如 10-12）开始测试。
ZOOM_LEVELS = [12,13,14,15,16,17,18,19] 

# 定义下载区域的经纬度边界框 (Bounding Box)
# 格式: [min_lng, min_lat, max_lng, max_lat]
# 下面是北京市中心的一个示例范围
BBOX = [114, 30, 115, 31] 

# 瓦片服务器 URL 模板 (OpenStreetMap)
# {s} 是子域名，用于负载均衡 (a, b, c)
# {z} 是层级, {x} 是x坐标, {y} 是y坐标
#TILE_URL_TEMPLATE = "https://{s}.tile.openstreetmap.org/{z}/{x}/{y}.png"
TILE_URL_TEMPLATE ="https://wprd01.is.autonavi.com/appmaptile?lang=zh_cn&size=1&style=7&x={x}&y={y}&z={z}"

# 保存瓦片的根目录
OUTPUT_DIR = "offline_tiles"
# --- 配置结束 ---


def deg2num(lat_deg, lon_deg, zoom):
    """
    将经纬度转换为瓦片坐标 (x, y)
    """
    lat_rad = math.radians(lat_deg)
    n = 2.0 ** zoom
    x_tile = int((lon_deg + 180.0) / 360.0 * n)
    y_tile = int((1.0 - math.asinh(math.tan(lat_rad)) / math.pi) / 2.0 * n)
    return x_tile, y_tile

def download_tiles():
    """
    根据配置下载并保存瓦片
    """
    min_lng, min_lat, max_lng, max_lat = BBOX

    for zoom in ZOOM_LEVELS:
        print(f"\n--- 开始下载层级 {zoom} ---")
        
        # 计算边界框四个角点对应的瓦片坐标
        x_min, y_min = deg2num(max_lat, min_lng, zoom)
        x_max, y_max = deg2num(min_lat, max_lng, zoom)
        
        total_tiles = (x_max - x_min + 1) * (y_max - y_min + 1)
        print(f"层级 {zoom} 需要下载 {total_tiles} 张瓦片...")

        # 使用 tqdm 创建进度条
        pbar = tqdm(total=total_tiles, unit="tile")

        for x in range(x_min, x_max + 1):
            for y in range(y_min, y_max + 1):
                # 构造瓦片的保存路径
                tile_dir = os.path.join(OUTPUT_DIR, str(zoom), str(x))
                os.makedirs(tile_dir, exist_ok=True)
                tile_path = os.path.join(tile_dir, f"{y}.png")

                # 如果文件已存在，则跳过下载
                if os.path.exists(tile_path):
                    pbar.update(1)
                    continue

                # 构造下载 URL
                # 使用随机子域名 {s}，这里我们循环使用 a, b, c
                subdomain = "abc"[(x + y) % 3]
                url = TILE_URL_TEMPLATE.format(s=subdomain, z=zoom, x=x, y=y)

                try:
                    # 下载瓦片
                    response = requests.get(url, timeout=10)
                    response.raise_for_status()  # 如果请求失败则抛出异常

                    # 保存瓦片
                    with open(tile_path, 'wb') as f:
                        f.write(response.content)
                    
                except requests.exceptions.RequestException as e:
                    print(f"\n下载失败: {url} -> {e}")
                    # 如果下载失败，可以选择删除空文件或记录日志
                    if os.path.exists(tile_path):
                        os.remove(tile_path)
                
                pbar.update(1)
        
        pbar.close()
        print(f"--- 层级 {zoom} 下载完成 ---")

    print("\n所有指定层级的瓦片下载完毕！")
    print(f"瓦片已保存至: {os.path.abspath(OUTPUT_DIR)}")


if __name__ == "__main__":
    download_tiles()